from datetime import datetime
from typing import Literal, List

from pydantic import BaseModel


class PredictionRequest(BaseModel):
    type: Literal['solar', 'wind', 'load','grid']  # 预测类型
    id: int

class PredictionOut(BaseModel):
    time_list: List[datetime]
    value_list: List[float]
